from selenium.webdriver.common.by import By
from base.base_action import BaseAction


# 对登录页面进行封装
class LoginPage(BaseAction):

    # 将所有用到的元素定位抽离出来
    # 用户名 输入框
    username_input = By.ID, "username"
    # 密码 输入框
    password_input = By.ID, "password"
    # 验证码 输入框
    verify_code_input = By.ID, "verify_code"
    # 登录 按钮
    login_btn = By.NAME, "sbtbutton"



    # 1.输入用户名
    def input_username(self, username):
        # 输入的2个传参：在哪输入（元素定位）、输入内容
        return self.input(self.username_input, username)

    # 2.输入密码
    def input_password(self, content):
        return self.input(self.password_input, content)

    # 3.输入验证码
    def input_verify_code(self, code):
        return self.input(self.verify_code_input, code)

    # 4.点击登录按钮
    def click_login_btn(self):
        return self.click(self.login_btn)



